1時間プログラミング 003 自作treeコマンド

概要

使い方

code:terminal

$ e1tree test

test

└── a

├── b

│ ├── f1

│ └── f2

└── c

└── f1

実装

子から親方向に向けて、再帰的に出力行を組み立てた

ディレクトリ階層毎に出力行を作り直すので効率は悪いが、実装が簡単

コードを一部抜粋

code:main.rs

fn build_tree_lines(path: &Path) -> Result<Vec<String>> {

let mut result = Vec::new();

let file_name = path.file_name().unwrap().to_str().unwrap().to_string();

result.push(file_name);

if path.is_dir() {

let mut entries = fs::read_dir(path)?

.map(|res| res.map(|e| e.path()))

.collect::<Result<Vec<_>, std::io::Error>>()?;

entries.sort();

for (i, path) in entries.iter().enumerate() {

let is_last_entry = i == entries.len() - 1;

let lines = build_tree_lines(path)?;

assert!(!lines.is_empty());

let (child, descendants) = lines.split_at(1);

// child

let prefix = if is_last_entry {

"└── "

} else {

"├── "

};

// descendants (excluded child)

let prefix = if is_last_entry { " " } else { "│ " };

for e in descendants {

result.push(format!("{}{}", prefix, e));

}

}

}

Ok(result)

}

fn main() -> Result<()> {

let opt = Opt::from_args();

let dir = Path::new(&opt.directory);

if !dir.is_dir() {

return Err(anyhow!("argument is not directory"));

}

let lines = build_tree_lines(dir)?;

for l in lines {

println!("{}", l);

}

Ok(())

}

感想

Rustでファイルシステム周りを操作した経験がなく、そこの調査にやや時間がかかった